package com.hrt.myprestocode;

import com.facebook.presto.spi.Plugin;
import com.google.common.collect.ImmutableSet;

import java.util.Set;


/**
 * Prsto 注册函数类，需要实现Plugin 接口
 */
public class MyFunctionPlugin implements Plugin {
    @Override
    public Set<Class<?>> getFunctions() {
        return ImmutableSet.<Class<?>>builder()
                //注册UDF，这里填写对应的UDF类
                .add(MyUDF.class)
                //注册UDAF函数，这里填写对应的UDAF类
                .add(MyUDAF.class)
                .build();
    }
}
